/* -*- C++ -*- */

/****************************************************************************
 ** Copyright (c) quickfixengine.org  All rights reserved.
 **
 ** This file is part of the QuickFIX FIX Engine
 **
 ** This file may be distributed under the terms of the quickfixengine.org
 ** license as defined by quickfixengine.org and appearing in the file
 ** LICENSE included in the packaging of this file.
 **
 ** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 ** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 **
 ** See http://www.quickfixengine.org/LICENSE for licensing information.
 **
 ** Contact ask@quickfixengine.org if any conditions of this licensing are
 ** not clear to you.
 **
 ****************************************************************************/

#ifndef FIX_SOCKETMONITOR_H
#define FIX_SOCKETMONITOR_H

#ifdef _MSC_VER
#pragma warning( disable : 4503 4355 4786 4290 )
#endif

#ifdef _MSC_VER
#include <Winsock2.h>
typedef int socklen_t;
#else
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#endif

#include <set>
#include <queue>
#include <time.h>

namespace FIX {
/// Monitors events on a collection of sockets.
class SocketMonitor {
public:
	class Strategy;

	SocketMonitor(int timeout = 0);
	virtual ~SocketMonitor();

	bool addConnect(int socket);
	bool addRead(int socket);
	bool addWrite(int socket);
	bool drop(int socket);
	void signal(int socket);
	void unsignal(int socket);
	void block(Strategy& strategy, bool poll = 0);

	size_t numSockets() {
		return m_readSockets.size() - 1;
	}

private:
	typedef std::set<int> Sockets;
	typedef std::queue<int> Queue;

	void setsockopt();
	bool bind();
	bool listen();
	void buildSet(const Sockets&, fd_set&);
	inline timeval* getTimeval(bool poll);
	inline bool sleepIfEmpty(bool poll);

	void processReadSet(Strategy&, fd_set&);
	void processWriteSet(Strategy&, fd_set&);
	void processExceptSet(Strategy&, fd_set&);

	int m_timeout;
	timeval m_timeval;
#ifndef SELECT_DECREMENTS_TIME
	clock_t m_ticks;
#endif

	int m_signal;
	int m_interrupt;
	Sockets m_connectSockets;
	Sockets m_readSockets;
	Sockets m_writeSockets;
	Queue m_dropped;

public:
	class Strategy {
	public:
		virtual ~Strategy() {
		}
		virtual void onConnect(SocketMonitor&, int socket) = 0;
		virtual void onEvent(SocketMonitor&, int socket) = 0;
		virtual void onWrite(SocketMonitor&, int socket) = 0;
		virtual void onError(SocketMonitor&, int socket) = 0;
		virtual void onError(SocketMonitor&) = 0;
		virtual void onTimeout(SocketMonitor&) {
		}
	};
};

/// Monitors events on a collection of sockets.
class SocketMonitorSenderReceiver {
public:
	class Strategy;

	SocketMonitorSenderReceiver(int timeout = 0);
	virtual ~SocketMonitorSenderReceiver();

	bool addConnect(int s, sockaddr_in &addr, ip_mreq &mreq);
	bool addRead(int socket);
	bool addWrite(int socket);
	bool drop(int socket);
	void signal(int socket);
	void unsignal(int socket);
	void block(Strategy& strategy, bool poll = 0);
	void blockRead(Strategy& strategy, bool poll = 0);
	void blockSend(Strategy& strategy, bool poll = 0);

	size_t numSockets() {
		return m_readSockets.size() - 1;
	}

	struct SocketInfo {
		SocketInfo() {
		}
		SocketInfo(sockaddr_in addr, ip_mreq mreq) :
			m_addr(addr), m_mreq(mreq) {
		}
		sockaddr_in m_addr;
		ip_mreq m_mreq;
	};
private:
	typedef std::map<int, SocketInfo> SocketInfoMap;
public:
	SocketInfo getSocketInfo(int s) {
		return m_socketInfoMap[s];
	}

private:
	typedef std::set<int> Sockets;
	typedef std::queue<int> Queue;

	void setsockopt();
	bool bind();
	bool listen();
	void buildSet(const Sockets&, fd_set&);
	inline timeval* getTimeval(bool poll);
	inline bool sleepIfEmpty(bool poll);

	void processReadSet(Strategy&, fd_set&);
	void processWriteSet(Strategy&, fd_set&);
	void processExceptSet(Strategy&, fd_set&);

	int m_timeout;
	timeval m_timeval;
#ifndef SELECT_DECREMENTS_TIME
	clock_t m_ticks;
#endif

	int m_signal;
	int m_interrupt;
	SocketInfoMap m_socketInfoMap;
	Sockets m_connectSockets;
	Sockets m_readSockets;
	Sockets m_writeSockets;
	Queue m_dropped;

public:
	class Strategy {
	public:
		virtual ~Strategy() {
		}
		virtual void onConnect(SocketMonitorSenderReceiver&, int socket) = 0;
		virtual void onEvent(SocketMonitorSenderReceiver&, int socket) = 0;
		virtual void onWrite(SocketMonitorSenderReceiver&, int socket) = 0;
		virtual void onError(SocketMonitorSenderReceiver&, int socket) = 0;
		virtual void onError(SocketMonitorSenderReceiver&) = 0;
		virtual void onTimeout(SocketMonitorSenderReceiver&) {
		}
	};
};
}

#endif //FIX_SOCKETMONITOR_H
